
%reload_ext autoreload
%autoreload 2

import sys
import os

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

plt.rcParams.update({'figure.max_open_warning': 0})


from viz.plotting.utils import resize_by_tradeoff, color_by_dist, color_by_cv, enhance_color, cfs

def load_props(path):
    r"""
    Load CV, Mu, Ik and S from path.
    """
    CV = None
    cvpathf = os.path.join(path, "cv.csv")
    if os.path.exists(cvpathf):
        CV = np.loadtxt(cvpathf)

    # load the Mu values
    Mu = None
    mupathf = os.path.join(path, "mu.csv")
    if os.path.exists(mupathf):
        Mu = np.loadtxt(mupathf)

    # load the knee indices
    Ik = None
    kpathf = os.path.join(path, "muid.csv")
    if os.path.exists(kpathf):
        Ik = np.loadtxt(kpathf, dtype = int)
        
    return CV, Mu, Ik

def decide_size_color(pf, dim, CV, Mu, Ik):
    r"""
    Decide point-size and color from `CV`, `Mu` and `Ik`
    """
    # decide point-size
    S = 1
    if Mu is not None and Ik is not None:
        S = resize_by_tradeoff(Mu, k=Ik)

    # decide colors
    # C = default_color(F.shape[0], alpha=0.5)
    M = np.mean(F, axis=0)
    C, D = color_by_dist(F, M)
    
    # if CV is available, use CV to color
    if CV is not None:
        pfname = pf.split('-')[0]
        if 'dtlz8' in pfname:
            C = color_by_cv(CV, factor=cfs[pfname][int(dim[:-1])])
        else:
            C = color_by_cv(CV)

    # enhance knee points
    if Ik is not None:
        C = enhance_color(C, Ik)
    return S, C, D


%matplotlib inline
from viz.plotting import scatter
from viz.plotting.scatter import camera_angles

pfs = {'dtlz2': ['3d', '4d', '8d'], \
       'dtlz2-nbi': ['3d', '4d', '5d', '8d'], \
       'debmdk': ['3d', '4d', '8d'], \
       'debmdk-nbi': ['3d', '4d', '8d'], \
       'debmdk-all': ['3d', '4d', '8d'], \
       'debmdk-all-nbi': ['3d', '4d', '8d'], \
       'dtlz8': ['3d', '4d', '6d', '8d'], \
       'dtlz8-nbi': ['3d', '4d', '6d', '8d'], \
       'c2dtlz2': ['3d', '4d', '5d', '8d'], \
       'c2dtlz2-nbi': ['3d', '4d', '5d', '8d'], \
       'cdebmdk': ['3d', '4d', '8d'], \
       'cdebmdk-nbi': ['3d', '4d', '8d'], \
       'c0dtlz2': ['3d', '4d', '8d'], \
       'c0dtlz2-nbi': ['3d', '4d', '8d'], \
       'crash-nbi': ['3d'], 'crash-c1-nbi': ['3d'], 'crash-c2-nbi': ['3d'], \
       'carside-nbi': ['3d'], \
       'gaa-old': ['10d'], 'gaa-old-nbi': ['10d'], \
       'gaa-fmincon': ['10d'], 'gaa-fmincon-nbi': ['10d'], \
       'gaa-nsga3': ['10d'], 'gaa-nsga3-nbi': ['10d']}

constrained_pfs = ['dtlz8', 'dtlz8-nbi', 'c2dtlz2', 'c2dtlz2-nbi', 'cdebmdk', 'cdebmdk-nbi', \
                   'c0dtlz2', 'c0dtlz2-nbi', 'carside-nbi', 'gaa-old', 'gaa-old-nbi', \
                   'gaa-fmincon', 'gaa-fmincon-nbi', 'gaa-nsga3', 'gaa-nsga3-nbi']

for pf in list(pfs.keys()):
    for dim in pfs[pf]:
        fullpathf = "../data/{0:s}/{1:s}/f.csv".format(pf, dim)
        if os.path.exists(fullpathf):
            path, filenamef = os.path.split(fullpathf)
            dirs = path.split('/')
            frontname = dirs[-2]

            # load the front
            F = np.loadtxt(fullpathf, delimiter=',')
            print(fullpathf, F.shape, dirs, frontname)
            
            # decide color and size
            CV, Mu, Ik = load_props(path)
            S,C,D = decide_size_color(pf, dim, CV, Mu, Ik)
            
            # make indices for knee and non-knee points
            I = np.zeros(F.shape[0]).astype(bool)
            I[Ik] = True
            Ip, Ik = ~I, I
            
            # fix point labels
            L = np.array(["" for _ in range(F.shape[0])]).astype(object)
            L[Ip] = "All Other Points"
            L[Ik] = "Knee Points"
            
            # decide what label and gradient to use for the colorbar
            if pf in constrained_pfs:
                cbl = "Normalized Cumulative CV"
                cbg = CV
            else:
                cbl = "Centroid Distance"
                cbg = D
                
            euler = camera_angles[pf][dim] if pf in camera_angles else [-60,30]
            
            figpath = os.path.join(path, "scatter3d.pdf")
            with plt.rc_context({"text.usetex": True, "font.size": 12}):
                # plot
                scatter.plot(F, ax=Axes3D(plt.figure()), s=S, c=C, labels=L, colorbar=(C[Ip],cbg[Ip],cbl), \
                             euler=euler)
                # save the fig
                plt.savefig(figpath, bbox_inches='tight', pad_inches=0, dpi=150)
                # show
                plt.show()

../data/dtlz2/3d/f.csv (1000, 3) ['..', 'data', 'dtlz2', '3d'] dtlz2

../data/dtlz2/4d/f.csv (2000, 4) ['..', 'data', 'dtlz2', '4d'] dtlz2

../data/dtlz2/8d/f.csv (4000, 8) ['..', 'data', 'dtlz2', '8d'] dtlz2

../data/dtlz2-nbi/3d/f.csv (990, 3) ['..', 'data', 'dtlz2-nbi', '3d'] dtlz2-nbi

../data/dtlz2-nbi/4d/f.csv (1771, 4) ['..', 'data', 'dtlz2-nbi', '4d'] dtlz2-nbi

../data/dtlz2-nbi/8d/f.csv (3432, 8) ['..', 'data', 'dtlz2-nbi', '8d'] dtlz2-nbi

../data/debmdk/3d/f.csv (1047, 3) ['..', 'data', 'debmdk', '3d'] debmdk

../data/debmdk/4d/f.csv (1967, 4) ['..', 'data', 'debmdk', '4d'] debmdk

../data/debmdk/8d/f.csv (4077, 8) ['..', 'data', 'debmdk', '8d'] debmdk

../data/debmdk-nbi/3d/f.csv (1143, 3) ['..', 'data', 'debmdk-nbi', '3d'] debmdk-nbi

Generators (Scatter Plots)¶

Notebook for generating basic scatter plots of the Pareto-optimal fronts with appropriate colorings and point-sizes.¶

Generate and save plots.¶